from litestar.contrib.sqlalchemy.dto import SQLAlchemyDTO
from litestar.contrib.sqlalchemy.repository import SQLAlchemyAsyncRepository
from litestar.dto import DTOConfig
from typing import Annotated
from models.test1 import Test1Model
# from sqlalchemy.ext.asyncio import AsyncSession

__all__ = ["WriteDTO", "ReadDTO"]

# class Repository(AsyncSession):
    # model_type = Test1Model

WriteDTO = SQLAlchemyDTO[Annotated[Test1Model, DTOConfig(exclude={"created_at", "updated_at"})]]
ReadDTO = SQLAlchemyDTO[Test1Model]